import string
import urllib.request

from urllib.parse import quote
from bs4 import BeautifulSoup
import pymysql
db = pymysql.connect(host='localhost',
                         port=3306,
                         user='root',
                         passwd='111111',
                         charset='utf8',
                         db='test')
cursor = db.cursor()
def insert(data):
    for job_dic in data:
        print(job_dic)
        cursor.execute("insert into job(job_name,job_sal,company_name,job_place,job_exp,job_education) values(%s,%s,%s,%s,%s,%s)",(job_dic["job_name"],job_dic["job_sal"],job_dic["company_name"],job_dic["job_place"],job_dic["job_exp"],job_dic["job_education"]))
    db.commit()

city=input("请输入城市: ")

code={
    '北京': '530',
    '上海': '538',
    '广州': '763',
    '深圳': '765',
    '天津': '531',
    '武汉': '736',
    '西安': '854',
    '成都': '801',
    '沈阳': '599',
    '南京': '635',
    '杭州': '653',
    '苏州': '639',
    '重庆': '551',
    '长沙': '749',
    '厦门': '682',
    '南昌': '691'
}

city=code[city]
type=input("请输入岗位:")
url="https://sou.zhaopin.com/?jl="+city+"&kw="+type+"&p=1"
url=quote(url,safe=string.printable)
res=urllib.request.urlopen(url)
print(url)
content=res.read().decode()
#print(content)
soup=BeautifulSoup(content,"lxml")
jobitem=soup.select(".joblist-box__item")

job_list=[]
for job in jobitem:
    job_dic={}
    job_name=job.select(".jobinfo__name")[0].get_text()
    job_sal=job.select(".jobinfo__salary")[0].get_text()
    job_sal=job_sal.strip().replace("","").replace("\n","")
    print(job_sal)
    if'天' in job_sal:
        #job_sal=int(job_sal.split("-"))[0]*30
        job_sal_list = job_sal.split("-")
        job_sal = int(job_sal_list[0]) * 30
    elif('千' in job_sal):
        job_sal=int(job_sal.split("千")[0])*1000
    elif('万'in job_sal):
        job_sal=float(job_sal.split("万")[0])*10000
    else:
        job_sal=5000
    company_name=job.select(".companyinfo__name")[0].get_text()
    job_msg=job.select(".jobinfo__other-info-item")
    job_place=job_msg[0].get_text().split('·')[0]
    job_exp=job_msg[1].get_text()
    job_education=job_msg[2].get_text()
    job_dic["job_name"]=job_name.strip()
    job_dic["job_sal"]=job_sal
    job_dic["company_name"]=company_name.strip()
    job_dic["job_place"]=job_place.strip()
    job_dic["job_exp"]=job_exp.strip()
    job_dic["job_education"]=job_education.strip()
    job_list.append(job_dic)
    # print(job_dic)
insert(job_list)